package com.raowb.common.config.asyn;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

/**
 * 
 * @Description: 异步调用配置
 * @author: raowb
 * @date 2018年8月12日
 */
@Configuration
@EnableAsync
public class AsyncConfig implements AsyncConfigurer {

	// 自定义线程池
	@Override
	public Executor getAsyncExecutor() {
		ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
		executor.setCorePoolSize(10);// 线程池创建时候初始化的线程数
		executor.setMaxPoolSize(50);// 线程池最大的线程数，只有在缓冲队列满了之后才会申请超过核心线程数的线程
		executor.setQueueCapacity(200); // 用来缓冲执行任务的队列
		executor.setKeepAliveSeconds(60);// 当超过了核心线程出之外的线程在空闲时间到达之后会被销毁
		executor.setThreadNamePrefix("AsyncThread-");
		// 线程池对拒绝任务的处理策略：这里采用了CallerRunsPolicy策略，当线程池没有处理能力的时候，该策略会直接在 execute
		// 方法的调用线程中运行被拒绝的任务；如果执行程序已关闭，则会丢弃该任务
		executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

		executor.setWaitForTasksToCompleteOnShutdown(true);// 设置线程池关闭的时候等待所有任务都完成再继续销毁其他的Bean
		executor.setAwaitTerminationSeconds(60);// 设置线程池中任务的等待时间，如果超过这个时候还没有销毁就强制销毁，以确保应用最后能够被关闭，而不是阻塞住

		executor.initialize(); // 如果不初始化，导致找到不到执行器
		return executor;
	}

	/**
	 * 无返回值的异常处理
	 */
	@Override
	public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
		return new AsyncExceptionHandler();
	}
}